import os
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

# 提示词模板
prompt = ChatPromptTemplate.from_template("出给一个关于 {goods} 的广告宣传语")
# ChatGPT模型调用对象
model = ChatOpenAI()
# 将两个对象使用顺序组合创建一个调用链，实现提示词组装，模型调用的功能
chain = prompt | model
# 输入提示词模版中的变量部分，调用链会自动完成后续的调用和解析
res = chain.invoke({"goods": "冰淇淋"})
print(res)
